using interview questions in pandas